package com.example.JavaWebApi.core.encryptionhelper;

import lombok.SneakyThrows;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;
import java.nio.charset.StandardCharsets;
import java.security.SecureRandom;
import java.util.Base64;

/**
 * @program: cms_zjgs_api
 * @description: 密码加密解密帮助类
 * @author: yang ze
 * @create: 2021-12-06 16:43
 */
@Component
public class EncryptionHelper {

    @Value("${aesconfig.key}")
    private String key;

    private byte[] getKeyByByte(String key){
        byte[] res=new byte[16];
        byte[] ori=key.getBytes(StandardCharsets.UTF_8);
        for(int i=0;i<=res.length-1;i++){
            if (i<=ori.length-1){
                res[i]=ori[i];
            }
            else {
                res[i]=0;
            }
        }
        return res;
    }
    /**
     * 生成密钥对象
     */
    private SecretKey generateKey(byte[] key) throws Exception {
        // 根据指定的 RNG 算法, 创建安全随机数生成器
        SecureRandom random = SecureRandom.getInstance("SHA1PRNG");
        // 设置 密钥key的字节数组 作为安全随机数生成器的种子
        random.setSeed(key);

        // 创建 AES算法生成器
        KeyGenerator gen = KeyGenerator.getInstance("AES");
        // 初始化算法生成器
        gen.init(128, random);

        // 生成 AES密钥对象, 也可以直接创建密钥对象: return new SecretKeySpec(key, ALGORITHM);
        return gen.generateKey();
    }
    /**
     * 数据加密: 明文 -> 密文
     */
    public String encrypt(String content) throws Exception {
        // 生成密钥对象
        SecretKey secKey = generateKey(getKeyByByte(key));
        byte[] plainBytes= content.getBytes(StandardCharsets.UTF_8);
        // 获取 AES 密码器
        Cipher cipher = Cipher.getInstance("AES");
        // 初始化密码器（加密模型）
        cipher.init(Cipher.ENCRYPT_MODE, secKey);
        // 加密数据, 返回密文
        return Base64.getEncoder().encodeToString(cipher.doFinal(plainBytes));
    }

    /**
     * 数据解密: 密文 -> 明文
     */
    public  String decrypt(String content) throws Exception {
        // 生成密钥对象
        SecretKey secKey = generateKey(getKeyByByte(key));
        byte[] cipherBytes=Base64.getDecoder().decode(content);
        // 获取 AES 密码器
        Cipher cipher = Cipher.getInstance("AES");
        // 初始化密码器（解密模型）
        cipher.init(Cipher.DECRYPT_MODE, secKey);

        // 解密数据, 返回明文
        return new String(cipher.doFinal(cipherBytes),StandardCharsets.UTF_8);
    }
}